from django.shortcuts import render,redirect
from app01 import models
from app01.utils.form import UserModelForm



def user_list(request):
    queryset=models.UserInfo.objects.all()
    return render(request,'user_list.html',{'queryset':queryset})

def user_add(request):
    if request.method=='GET':
        form=UserModelForm()
        return  render(request,'user_add.html',{'form':form})
    #对提交的数据进行校验处理
    form=UserModelForm(data=request.POST)
    # 如果校验成功就保存
    if form.is_valid():
        print(form.cleaned_data)
        form.save()
        return redirect('/user/list/')
    #如果校验失败，就返回页面
    return render(request,'user_add.html',{'form':form})

def user_delete(request,nid):
    models.UserInfo.objects.filter(id=nid).delete()
    return redirect('/user/list/')


def user_edit(request,nid):
    # 根据ID查找数据库的员工信息
    row_objet = models.UserInfo.objects.filter(id=nid).first()
    if request.method=='GET':
        # 在form中把数据展示出来
        form=UserModelForm(instance=row_objet)
        return render(request,'user_edit.html',{'form':form})

    # 把数据库的数据传到form进行更新
    form = UserModelForm(data=request.POST,instance=row_objet)
    if form.is_valid():
        # 用户输入额外的值想保存
        # form.instance.数据库字段名=值
        # 保存数据输入的值保存
        form.save()
        return redirect('/user/list/')
    return render(request,'user_edit.html',{'form':form})



